import argparse


def parse_args():
    parser = argparse.ArgumentParser(description="catboost")

    parser.add_argument('--epoch', type=int, default=1000, help='number of epochs')
    parser.add_argument('--run_id', type=int, default=1)
    parser.add_argument('--depth', type=int, default=7)
    parser.add_argument('--l2_leaf_reg', type=int, default=3)


    return parser.parse_args()
